#创建一个驱动对象，避免频繁创建驱动
from selenium import webdriver
import datetime
import os.path
import sys

class Driver:
    driver=""
    def __init__(self):
        #这里可以添加浏览器参数（例如：可以设置 无头模式）
        options=webdriver.ChromeOptions()
        path = "C:\\Users\\Hasee\\Downloads\\chromedriver-win64\\chromedriver.exe"
        self.driver = webdriver.Chrome(executable_path=path,options=options)

    def getScreenShot(self):
        dirname=datetime.datetime.now().strftime('%Y-%m-%d')
        #判断当前路径是否存在，不存在则创建文件夹
        if not os.path.exists("../images/"+dirname):
            os.mkdir("../images/"+dirname)
        filename=sys._getframe().f_back.f_code.co_name+"-"+datetime.datetime.now().strftime('%Y-%m-%d-%H%M%S')+".png"
        self.driver.save_screenshot('../images/'+dirname+'/'+filename)

BlogDriver=Driver()


